package utils;

import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import org.apache.commons.codec.binary.Base64;
/**
 * Description：用此类将图片转换为字符串，以便将图片封装为JSON进行传输
 * @author
 * @Date
 * @version 1.0
 * */
public class ImgUtil {

    /**
     * TODO:将byte数组以Base64方式编码为字符串
     * @param bytes 待编码的byte数组
     * @return 编码后的字符串
     * */
    public static String encode(byte[] bytes){
        return Base64.encodeBase64String(bytes);
    }

    /**
     * TODO:将以Base64方式编码的字符串解码为byte数组
     * @param encodeStr 待解码的字符串
     * @return 解码后的byte数组
     * @throws IOException
     * */
    public static byte[] decode(String encodeStr){
        byte[] bt = null;
        bt =Base64.decodeBase64(encodeStr);
        return bt;
    }

    /**
     * TODO:将两个byte数组连接起来后，返回连接后的Byte数组
     * @param front 拼接后在前面的数组
     * @param after 拼接后在后面的数组
     * @return 拼接后的数组
     * */
    public static byte[] connectBytes(byte[] front, byte[] after){
        byte[] result = new byte[front.length + after.length];
        System.arraycopy(front, 0, result, 0, after.length);
        System.arraycopy(after, 0, result, front.length, after.length);
        return result;
    }

    /**
     * TODO:将图片以Base64方式编码为字符串
     * @param imgUrl 图片的绝对路径（例如：D:\\jsontest\\abc.jpg）
     * @return 编码后的字符串
     * @throws IOException
     * */
    public static String encodeImage(String imgUrl) throws IOException{
        FileInputStream fis = new FileInputStream(imgUrl);
        byte[] rs = new byte[fis.available()];
        fis.read(rs);
        fis.close();
        return encode(rs);
    }

    /**
     * 将字符串自动解码为byte数组并写入图片
     * @param encodeStr 编码的字符串
     * @param imgUrl 图片的路径
     * @throws IOException
     */
    public static void loadImage(String encodeStr,String imgUrl ) throws IOException {
        byte[] b= decode(encodeStr);
        FileOutputStream fos=new FileOutputStream(imgUrl);
        fos.write(b);
        fos.close();
    }

    /**
     * @param args
     */
    public static void main(String[] args) throws IOException {
        String str = null;
        try {
            str = encodeImage("src/1.png");
        } catch (IOException e) {
            e.printStackTrace();
        }


        byte[] b= decode(str);
        FileOutputStream fos=new FileOutputStream("src/2.png");
        fos.write(b);

    }
}